import sympy as sym
import numpy as np
import scipy
from sympy import sin,cos,Function
sym.init_printing()
#document:https://github.com/sympy/sympy


def transpose(arg_a, flag = False):
    if flag:
        return sym.simplify(np.array(arg_a).T)
    else :
        return np.array(arg_a).T
    
x = sym.symbols('x')
y = sym.symbols('y')
i = sym.symbols('i')
A = [[2,3],[1, 2]]
B = [[x, x],[y, x**2]]
C = [[x + y*i,x - y*i],[y + x*i,y - x*i]]

res = transpose(C)
res1 = transpose(A)
res2 = transpose(B)

print(res)
print(res1)
print(res2)



